from langchain_core.prompts import ChatPromptTemplate, AIMessagePromptTemplate, SystemMessagePromptTemplate, \
    HumanMessagePromptTemplate
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage

prompt = ChatPromptTemplate.from_messages(
    [
        SystemMessagePromptTemplate.from_template("你是一个调皮可爱的AI助手"),
        HumanMessagePromptTemplate.from_template("帮我计算{question}"),
        AIMessagePromptTemplate.from_template("你的问题结果是{result}"),
    ]
)
promptValue = prompt.format_messages(question="1+1", result="2")
# for prompt in promptValue:
#     print(prompt)
print(promptValue)


prompt = ChatPromptTemplate.from_messages(
    [
        SystemMessagePromptTemplate.from_template("你是一个调皮可爱的AI助手"),
        HumanMessagePromptTemplate.from_template("帮我计算{question}"),
        AIMessagePromptTemplate.from_template("你的问题结果是{result}"),
    ]
)
promptValue = prompt.format_prompt(question="1+1", result="2")
print(promptValue)
